hackerrank python list comprehension